#include "../server.hpp"
#include <fstream>
#include <sys/stat.h>
#include <regex>
// 放在全局，不需要每次进入函数都构造一次
std::unordered_map<std::string, std::string> _mime_msg = {{".aac", "audio/aac"},
                                                          {".abw", "application/x-abiword"},
                                                          {".arc", "application/x-freearc"},
                                                          {".avi", "video/x-msvideo"},
                                                          {".azw", "application/vnd.amazon.ebook"},
                                                          {".bin", "application/octet-stream"},
                                                          {".bmp", "image/bmp"},
                                                          {".bz", "application/x-bzip"},
                                                          {".bz2", "application/x-bzip2"},
                                                          {".csh", "application/x-csh"},
                                                          {".css", "text/css"},
                                                          {".csv", "text/csv"},
                                                          {".doc", "application/msword"},
                                                          {".docx", "application/vnd.openxmlformats-officedocument.wordprocessingml.document"},
                                                          {".eot", "application/vnd.ms-fontobject"},
                                                          {".epub", "application/epub+zip"},
                                                          {".gif", "image/gif"},
                                                          {".htm", "text/html"},
                                                          {".html", "text/html"},
                                                          {".ico", "image/vnd.microsoft.icon"},
                                                          {".ics", "text/calendar"},
                                                          {".jar", "application/java-archive"},
                                                          {".jpeg", "image/jpeg"},
                                                          {".jpg", "image/jpeg"},
                                                          {".js", "text/javascript"},
                                                          {".json", "application/json"},
                                                          {".jsonld", "application/ld+json"},
                                                          {".mid", "audio/midi"},
                                                          {".midi", "audio/x-midi"},
                                                          {".mjs", "text/javascript"},
                                                          {".mp3", "audio/mpeg"},
                                                          {".mpeg", "video/mpeg"},
                                                          {".mpkg", "application/vnd.apple.installer+xml"},
                                                          {".odp", "application/vnd.oasis.opendocument.presentation"},
                                                          {".ods", "application/vnd.oasis.opendocument.spreadsheet"},
                                                          {".odt", "application/vnd.oasis.opendocument.text"},
                                                          {".oga", "audio/ogg"},
                                                          {".ogv", "video/ogg"},
                                                          {".ogx", "application/ogg"},
                                                          {".otf", "font/otf"},
                                                          {".png", "image/png"},
                                                          {".pdf", "application/pdf"},
                                                          {".ppt", "application/vnd.ms-powerpoint"},
                                                          {".pptx", "application/vnd.openxmlformats-officedocument.presentationml.presentation"},
                                                          {".rar", "application/x-rar-compressed"},
                                                          {".rtf", "application/rtf"},
                                                          {".sh", "application/x-sh"},
                                                          {".svg", "image/svg+xml"},
                                                          {".swf", "application/x-shockwave-flash"},
                                                          {".tar", "application/x-tar"},
                                                          {".tif", "image/tiff"},
                                                          {".tiff", "image/tiff"},
                                                          {".ttf", "font/ttf"},
                                                          {".txt", "text/plain"},
                                                          {".vsd", "application/vnd.visio"},
                                                          {".wav", "audio/wav"},
                                                          {".weba", "audio/webm"},
                                                          {".webm", "video/webm"},
                                                          {".webp", "image/webp"},
                                                          {".woff", "font/woff"},
                                                          {".woff2", "font/woff2"},
                                                          {".xhtml", "application/xhtml+xml"},
                                                          {".xls", "application/vnd.ms-excel"},
                                                          {".xlsx", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"},
                                                          {".xml", "application/xml"},
                                                          {".xul", "application/vnd.mozilla.xul+xml"},
                                                          {".zip", "application/zip"},
                                                          {".3gp", "video/3gpp"},
                                                          {".3g2", "video/3gpp2"},
                                                          {".7z", "application/x-7z-compressed"}};

std::unordered_map<int, std::string> _statu_msg = {{100, "Continue"},
                                                   {101, "Switching Protocol"},
                                                   {102, "Processing"},
                                                   {103, "Early Hints"},
                                                   {200, "OK"},
                                                   {201, "Created"},
                                                   {202, "Accepted"},
                                                   {203, "Non-Authoritative Information"},
                                                   {204, "No Content"},
                                                   {205, "Reset Content"},
                                                   {206, "Partial Content"},
                                                   {207, "Multi-Status"},
                                                   {208, "Already Reported"},
                                                   {226, "IM Used"},
                                                   {300, "Multiple Choice"},
                                                   {301, "Moved Permanently"},
                                                   {302, "Found"},
                                                   {303, "See Other"},
                                                   {304, "Not Modified"},
                                                   {305, "Use Proxy"},
                                                   {306, "unused"},
                                                   {307, "Temporary Redirect"},
                                                   {308, "Permanent Redirect"},
                                                   {400, "Bad Request"},
                                                   {401, "Unauthorized"},
                                                   {402, "Payment Required"},
                                                   {403, "Forbidden"},
                                                   {404, "Not Found"},
                                                   {405, "Method Not Allowed"},
                                                   {406, "Not Acceptable"},
                                                   {407, "Proxy Authentication Required"},
                                                   {408, "Request Timeout"},
                                                   {409, "Conflict"},
                                                   {410, "Gone"},
                                                   {411, "Length Required"},
                                                   {412, "Precondition Failed"},
                                                   {413, "Payload Too Large"},
                                                   {414, "URI Too Long"},
                                                   {415, "Unsupported Media Type"},
                                                   {416, "Range Not Satisfiable"},
                                                   {417, "Expectation Failed"},
                                                   {418, "I'm a teapot"},
                                                   {421, "Misdirected Request"},
                                                   {422, "Unprocessable Entity"},
                                                   {423, "Locked"},
                                                   {424, "Failed Dependency"},
                                                   {425, "Too Early"},
                                                   {426, "Upgrade Required"},
                                                   {428, "Precondition Required"},
                                                   {429, "Too Many Requests"},
                                                   {431, "Request Header Fields Too Large"},
                                                   {451, "Unavailable For Legal Reasons"},
                                                   {501, "Not Implemented"},
                                                   {502, "Bad Gateway"},
                                                   {503, "Service Unavailable"},
                                                   {504, "Gateway Timeout"},
                                                   {505, "HTTP Version Not Supported"},
                                                   {506, "Variant Also Negotiates"},
                                                   {507, "Insufficient Storage"},
                                                   {508, "Loop Detected"},
                                                   {510, "Not Extended"},
                                                   {511, "Network Authentication Required"}};
#define DEFAULT_TIMEOUT 10
class Util
{
public:
    // 字符串分割函数,将src字符串按照sep字符进行分割，得到的各个子串放到array中，最终返回子串的数量
    static size_t Split(const std::string &src, const std::string &sep, std::vector<std::string> *array)
    {
        size_t offset = 0;
        while (offset < src.size())
        {
            size_t pos = src.find(sep, offset); // 在src字符串偏移量offset处，开始向后查找sep字符/子串，返回查找到的位置
            if (pos == std::string::npos)
            {
                // 没有找到特定字符
                array->push_back(src.substr(offset));
                return array->size();
            }
            if (pos == offset) // 跳过连续的分割符
            {
                offset = pos + sep.size();
                continue; // 当前子串是一个空串，跳过
            }
            array->push_back(src.substr(offset, pos - offset));
            offset = pos + sep.size();
        }
        return array->size();
    }
    // 读取文件内容，将读取的内容放入一个buffer中
    static bool ReadFile(const std::string &filename, std::string *buffer)
    {
        std::ifstream ifs(filename, std::ios::binary); // 以二进制方式打开文件
        if (ifs.is_open() == false)
        {
            LOG(LogLevel::ERROR) << "open file " << filename << " failed";
            return false;
        }
        size_t fsize = 0;
        ifs.seekg(0, ifs.end); // 跳转读写位置到末尾
        fsize = ifs.tellg();   // 获取当前读写位置相对于起始位置的偏移量，从末尾偏移刚好就是文件大小
        ifs.seekg(0, ifs.beg); // 跳转到起始位置
        buffer->resize(fsize); // 预分配空间
        ifs.read(&((*buffer)[0]), fsize);
        if (ifs.good() == false)
        {
            LOG(LogLevel::ERROR) << "read file " << filename << " failed";
            ifs.close();
            return false;
        }
        ifs.close();
        return true;
    }
    // 向文件写入数据
    static bool WriteFile(const std::string &filename, const std::string &buf)
    {
        std::ofstream ofs(filename, std::ios::binary | std::ios::trunc); // 以二进制方式打开文件，并清空原有内容
        if (ofs.is_open() == false)
        {
            LOG(LogLevel::ERROR) << "open file " << filename << " failed";
            return false;
        }
        ofs.write(buf.c_str(), buf.size());
        if (ofs.good() == false)
        {
            LOG(LogLevel::ERROR) << "write file " << filename << " failed";
            ofs.close();
            return false;
        }
        ofs.close();
        return true;
    }
    // URL编码,避免URL中资源路径与查询字符串中的特殊字符与HTTP请求中特殊字符产生歧义
    // 编码格式：将特殊字符的ascii值，转化为两个16进制字符，前缀是%
    // W3C标准中规定，查询字符串中的空格，需要被编码为+，解码则是+转空格
    static std::string UrlEncode(const std::string url, bool convert_space_to_plus)
    {
        std::string res;
        for (auto &c : url)
        {
            if (c == '.' || c == '-' || c == '_' || c == '~' || isalnum(c)) // 如果是不需要编码的特殊字符或者本身就是数字或字母，就直接追加
            {
                res += c;
                continue;
            }
            if (c == ' ' && convert_space_to_plus == true)
            {
                res += '+';
                continue;
            }
            // 剩下的字符都是需要编码成为 %HH格式的
            char tmp[4] = {0};
            snprintf(tmp, 4, "%%%02X", c);
            res += tmp;
        }
        return res;
    }
    static int HEXTOI(char c)
    {
        if (c >= '0' && c <= '9')
            return c - '0';
        else if (c >= 'a' && c <= 'f')
            return c - 'a' + 10;
        return -1;
    }
    // URL解码
    static std::string UrlDecode(const std::string &url, bool convert_plus_to_space)
    {
        std::string res;
        // 遇到了%，则将紧随其后的2个字符，转换为数字，第一个数字左移4位(即*16)，然后加上第二个数字
        for (int i = 0; i < url.size(); i++)
        {
            if (url[i] == '+' && convert_plus_to_space == true)
            {
                res += ' ';
                continue;
            }
            if (url[i] == '%' && i + 2 < url.size())
            {
                int v1 = HEXTOI(url[i + 1]);
                int v2 = HEXTOI(url[i + 2]);
                char c = (v1 << 4) + v2;
                res += c;
                i += 2; // 跳过已经解码的两个字符
                continue;
            }
            res += url[i];
        }
        return res;
    }
    // 响应状态码的描述信息获取
    static std::string StatusDesc(int statu)
    {
        auto it = _statu_msg.find(statu);
        if (it != _statu_msg.end())
        {
            return it->second;
        }
        return "Unknown";
    }
    // 根据文件后缀名获取文件mime
    static std::string ExtMime(const std::string &filename)
    {
        // 先获取文件扩展名
        size_t pos = filename.find_last_of('.');
        if (pos == std::string::npos)
        {
            return "application/octet-stream"; // 没有后缀名，默认是二进制文件流
        }
        // 根据扩展名，获取mime
        std::string ext = filename.substr(pos);
        auto it = _mime_msg.find(ext);
        if (it != _mime_msg.end())
        {
            return it->second;
        }
        return "application/octet-stream";
    }
    // 判断文件是否是一个目录
    static bool IsDirect(const std::string &filename)
    {
        struct stat st;
        int ret = stat(filename.c_str(), &st);
        if (ret < 0) // 文件可能不存在
        {
            return false;
        }
        return S_ISDIR(st.st_mode);
    }
    // 判断文件是否是一个普通文件
    static bool IsRegular(const std::string &filename)
    {
        struct stat st;
        int ret = stat(filename.c_str(), &st);
        if (ret < 0) // 文件可能不存在
        {
            return false;
        }
        return S_ISREG(st.st_mode);
    }
    // http请求的资源路径的有效性判断
    // /index.html ---前面的/叫做相对根目录， 映射的是某个服务器上的子目录
    // 想表达的意思是，客户端只能请求相对根目录中的资源，其他地方的资源都不予理会
    static bool ValidPath(const std::string &path)
    {
        // 思想：按照/进行路径分割，根据有多少子目录，计算目录深度，有多少层，深度不能小于0
        std::vector<std::string> subdir;
        Split(path, "/", &subdir);
        int level = 0;
        for (auto &dir : subdir)
        {
            if (dir == "..")
            {
                level--; // 任意一层走出相对根目录，就认为有问题
                if (level < 0)
                {
                    return false;
                }
            }
            else if (dir == ".")
            {
                continue;
            }
            else
            {
                level++;
            }
        }
        return true;
    }
};

class HttpRequest
{
public:
    HttpRequest() : _version("HTTP/1.1") {}
    // 插入头部字段
    void SetHeader(const std::string &key, const std::string &value)
    {
        _headers.insert(std::make_pair(key, value));
    }
    // 判断是否存在某个头部字段
    bool HasHeader(const std::string &key) const
    {
        auto it = _headers.find(key);
        return it != _headers.end();
    }
    // 获取某个头部字段的值
    std::string GetHeader(const std::string &key) const
    {
        auto it = _headers.find(key);
        if (it != _headers.end())
        {
            return it->second;
        }
        return "";
    }
    // 插入查询字符串
    void SetParam(const std::string &key, const std::string &value)
    {
        _params.insert(std::make_pair(key, value));
    }
    // 判断是否存在某个指定的查询字符串
    bool HasParam(const std::string &key)
    {
        auto it = _params.find(key);
        return it != _params.end();
    }
    // 获取指定的查询字符串
    std::string GetParam(const std::string &key)
    {
        auto it = _params.find(key);
        if (it != _params.end())
        {
            return it->second;
        }
        return "";
    }
    // 获取正文长度
    size_t ContentLength()
    {
        bool ret = HasHeader("Content-Length");
        if (ret == false)
        {
            return 0;
        }
        std::string clen = GetHeader("Content-Length");
        return std::stol(clen); // string转long
    }
    // 判断是否是短连接
    bool Close() const
    {
        // 没有Connection字段，或者有Connection字段，但值为close，则是短连接，否则就是长连接
        if (HasHeader("Connection") == true && GetHeader("Connection") == "keep-alive")
        {
            // 长连接
            return false;
        }
        // 短连接
        return true;
    }
    // 将来会放在上下文中，每一次上下文里面的请求被处理完了，上下文就要被重置一下
    // 如果不重置，当前的请求信息就会对下一次的请求造成影响
    void ReSet()
    {
        _method.clear();
        _path.clear();
        _version = "HTTP/1.1";
        _body.clear();
        std::smatch match;
        _matches.swap(match);
        _headers.clear();
        _params.clear();
    }

public:
    std::string _method;                                   // 请求方法
    std::string _path;                                     // 资源路径
    std::string _version;                                  // http版本
    std::string _body;                                     // 正文
    std::smatch _matches;                                  // 资源路径的正则提取数据
    std::unordered_map<std::string, std::string> _headers; // 请求头部
    std::unordered_map<std::string, std::string> _params;  // 请求参数
};

class HttpResponse
{
public:
    HttpResponse() : _redirect_flag(false), _statu(200) {}
    HttpResponse(int statu) : _statu(statu), _redirect_flag(false) {}
    void ReSet()
    {
        _statu = 200;
        _redirect_flag = false;
        _redirect_url.clear();
        _body.clear();
        _headers.clear();
    }
    void SetHeader(const std::string &key, const std::string &value)
    {
        _headers.insert(std::make_pair(key, value));
    }
    bool HasHeader(const std::string &key)
    {
        auto it = _headers.find(key);
        return it != _headers.end();
    }
    std::string GetHeader(const std::string &key)
    {
        auto it = _headers.find(key);
        if (it != _headers.end())
        {
            return it->second;
        }
        return "";
    }
    void SetContent(const std::string &body, const std::string &content_type = "text/html")
    {
        _body = body;
        SetHeader("Content-Type", content_type);
    }
    void SetRedirect(const std::string &url, int statu = 302)
    {
        _statu = statu;
        _redirect_flag = true;
        _redirect_url = url;
    }
    bool Close()
    {
        if (HasHeader("Connection") == true && GetHeader("Connection") == "keep-alive")
        {
            // 长连接
            return false;
        }
        // 短连接
        return true;
    }

public:
    int _statu;                                            // 响应状态码
    bool _redirect_flag;                                   // 是否有重定向
    std::string _redirect_url;                             // 重定向的url
    std::string _body;                                     // 正文
    std::unordered_map<std::string, std::string> _headers; // 头部信息
};
typedef enum
{
    RECV_HTTP_ERROR,
    RECV_HTTP_LINE,
    RECV_HTTP_HEAD,
    RECV_HTTP_BODY,
    RECV_HTTP_OVER
} HttpRecvStatu;

#define MAX_LINE 8192
class HttpContext
{
public:
    HttpContext() {}
    int ResponseStatu()
    {
        return _resp_statu;
    }
    HttpRecvStatu RecvStatu()
    {
        return _recv_statu;
    }
    HttpRequest &Request()
    {
        return _request;
    }
    void RecvHttpRequest(Buffer *buf) // 接收并解析http请求
    {
        switch (_recv_statu)
        {
        // 不同的状态做不同的事，注意这里不需要break，就是要它一个一个往下执行，如果中间出错了也不用担心
        // 因为出错的话，_recv_statu会被置为RECV_HTTP_ERROR，哪怕走到下面的case了
        // 因为每个函数里都有对阶段的校验逻辑，所以如果阶段不对，就会从函数中退出，不会往下执行
        // 例如：处理完请求行之后，应该立即处理头部，而不是退出
        case RECV_HTTP_LINE:
            RecvHttpLine(buf);
        case RECV_HTTP_HEAD:
            RecvHttpHead(buf);
        case RECV_HTTP_BODY:
            RecvHttpBody(buf);
        }
        return;
    }
    void ReSet()
    {
        _recv_statu = RECV_HTTP_LINE;
        _request.ReSet();
        _resp_statu = 200; // 默认响应状态码为200
    }

private:
    bool ParseHttpLine(const std::string &line)
    {
        std::smatch matches;
        std::regex e("(GET|POST|HEAD|PUT|DELETE) ([^?]*)(?:\\?(.*))? (HTTP/1\\.[01])(?:\n|\r\n?)", std::regex::icase); // 后面这个是为了忽略大小写
        bool ret = std::regex_match(line, matches, e);
        if (ret == false)
        {
            _recv_statu = RECV_HTTP_ERROR;
            _resp_statu = 400; // 请求语法错误
            LOG(LogLevel::DEBUG) << _resp_statu;
            return false;
        }
        // 请求方法的获取
        _request._method = matches[1];
        //将请求方法转成大写
        std::transform(_request._method.begin(), _request._method.end(), _request._method.begin(), ::toupper);
        // 资源路径的获取，需要进行Url解码操作，但是这里不需要+转空格
        _request._path = Util::UrlDecode(matches[2], false);
        // 协议版本的获取
        _request._version = matches[4];
        // 查询字符串的获取与处理
        std::vector<std::string> query_string_arry;
        std::string query_string = matches[3];
        // 查询字符串的格式 key=value&key=value.....,先以 & 进行分割，得到各个子串
        Util::Split(query_string, "&", &query_string_arry);
        // 针对各个子串 以 = 进行分割，得到key和val，得到之后也需要进行URL解码
        for (auto &str : query_string_arry)
        {
            size_t pos = str.find('=');
            if (pos == std::string::npos)
            {
                _recv_statu = RECV_HTTP_ERROR;
                _resp_statu = 400; // 请求语法错误
                LOG(LogLevel::DEBUG) << _resp_statu;

                return false;
            }
            std::string key = Util::UrlDecode(str.substr(0, pos), false);
            // 在W3C标准中，查询字符串中空格是被转化成‘+’的，所以解码要转回来
            std::string value = Util::UrlDecode(str.substr(pos + 1), true);
            _request.SetParam(key, value);
        }
        return true;
    }
    bool RecvHttpLine(Buffer *buf)
    {
        if (_recv_statu != RECV_HTTP_LINE)
            return false;
        // 1.获取一行数据
        std::string line = buf->GetLineAndPop();
        // 2.需要考虑一些要素：缓冲区中的数据不足一行，获取的一行数据超大
        if (line.size() == 0)
        {
            // 缓冲区中的数据不足一行，则需要判断缓冲区的可读数据长度，如果很长了都不足一行，这是有问题的
            if (buf->ReadableSize() > MAX_LINE)
            {
                _recv_statu = RECV_HTTP_ERROR;
                _resp_statu = 414; // 请求的URI过长
                return false;
            }
            // 缓冲区中数据不足一行，但是也不多，就等等新数据的到来
            return true;
        }
        if (line.size() > MAX_LINE)
        {
            _recv_statu = RECV_HTTP_ERROR;
            _resp_statu = 414; // 请求的URI过长
            return false;
        }
        bool ret = ParseHttpLine(line);
        if (ret == false)
            return false;
        // 首行处理完毕，进入头部获取阶段
        _recv_statu = RECV_HTTP_HEAD;
        return true;
    }

    bool RecvHttpHead(Buffer *buf)
    {
        if (_recv_statu != RECV_HTTP_HEAD)
            return false;
        // 一行一行取出数据，直到遇到空行为止，头部的格式 key: val\r\n
        while (1)
        {
            std::string line = buf->GetLineAndPop(); // 这是带换行的
            // 2.需要考虑一些要素：缓冲区中的数据不足一行，获取的一行数据超大
            if (line.size() == 0)
            {
                // 缓冲区中的数据不足一行，则需要判断缓冲区的可读数据长度，如果很长了都不足一行，这是有问题的
                if (buf->ReadableSize() > MAX_LINE)
                {
                    _recv_statu = RECV_HTTP_ERROR;
                    _resp_statu = 414; // 请求的URI过长
                    return false;
                }
                // 缓冲区中数据不足一行，但是也不多，就等等新数据的到来
                return true;
            }
            if (line.size() > MAX_LINE)
            {
                _recv_statu = RECV_HTTP_ERROR;
                _resp_statu = 414; // 请求的URI过长
                return false;
            }
            if (line == "\n" || line == "\r\n")
            {
                // 头部接收完毕
                break;
            }
            bool ret = ParseHttpHead(line);
            if (ret == false)
                return false;
        }
        // 头部处理完毕，进入正文的接收阶段
        _recv_statu = RECV_HTTP_BODY;
        return true;
    }
    bool ParseHttpHead(std::string &line)
    {
        if (line.back() == '\n')
            line.pop_back(); // 去掉末尾的换行符
        if (line.back() == '\r')
            line.pop_back(); // 去掉末尾的回车符
        size_t pos = line.find(": ");
        if (pos == std::string::npos)
        {
            _recv_statu = RECV_HTTP_ERROR;
            _resp_statu = 400; // 请求语法错误
            return false;
        }
        std::string key = line.substr(0, pos);
        std::string value = line.substr(pos + 2);
        _request.SetHeader(key, value);
        return true;
    }
    bool RecvHttpBody(Buffer *buf)
    {
        // 1.获取正文长度
        size_t content_length = _request.ContentLength();
        if (content_length == 0)
        {
            // 没有正文，则请求接收完毕
            _recv_statu = RECV_HTTP_OVER;
            return true;
        }
        // 2.当前接收了多少正文，其实就是往_request._body放了多少数据了
        size_t real_len = content_length - _request._body.size(); // 实际还需要接收的正文长度

        // 3.接收正文放到body中，但是也要考虑当前缓冲区中的数据，是否是全部的正文
        // 3.1缓冲区中数据，包含了当前请求所有的正文，则取出所需的数据
        if (real_len <= buf->ReadableSize())
        {
            _request._body.append(buf->ReadPosition(), real_len);
            buf->MoveReadOffset(real_len);
            _recv_statu = RECV_HTTP_OVER;
            return true;
        }
        // 3.2缓冲区中数据，无法满足当前的正文需要，数据不足，取出数据，然后等待新数据到来
        _request._body.append(buf->ReadPosition(), buf->ReadableSize());
        buf->MoveReadOffset(buf->ReadableSize());
        return true;
    }

public:
    int _resp_statu = 200;                      // 响应状态码
    HttpRecvStatu _recv_statu = RECV_HTTP_LINE; // 当前接收及解析的阶段状态
    HttpRequest _request;                       // 已经解析得到的请求信息
};

class HttpServer
{
    using Handler = std::function<void(const HttpRequest &, HttpResponse *)>;
    using Handlers = std::vector<std::pair<std::regex, Handler>>; // 存储编译完毕的正则表达式
public:
    HttpServer(int port, int timeout = DEFAULT_TIMEOUT) : _server(port)
    {
        _server.EnableInactiveRelease(timeout);
        _server.SetConnectedCallback(std::bind(&HttpServer::OnConnected, this, std::placeholders::_1));
        _server.SetMessageCallback(std::bind(&HttpServer::OnMessage, this, std::placeholders::_1, std::placeholders::_2));
    }
    void SetBasDir(const std::string &path)
    {
        assert(Util::IsDirect(path) == true);
        _basedir = path;
    }
    // 设置/添加，请求(请求的正则表达式)与处理函数的映射关系
    void Get(const std::string &pattern, const Handler &handler)
    {
        // 资源路径不能忽略大小写
        _get_route.push_back(std::make_pair(std::regex(pattern), handler));
    }
    void Post(const std::string &pattern, const Handler &handler)
    {
        _post_route.push_back(std::make_pair(std::regex(pattern), handler));
    }
    void Put(const std::string &pattern, const Handler &handler)
    {
        _put_route.push_back(std::make_pair(std::regex(pattern), handler));
    }
    void Delete(const std::string &pattern, const Handler &handler)
    {
        _delete_route.push_back(std::make_pair(std::regex(pattern), handler));
    }
    void SetThreadCount(int count)
    {
        _server.SetThreadCount(count);
    }
    void Lsiten()
    {
        _server.Start();
    }

private:
    void ErrorHandler(const HttpRequest &req, HttpResponse *rsp)
    {
        // 1.组织一个错误展示页面
        std::string body;
        body += "<html>";
        body += "<head>";
        body += "meta http-equiv='Content-Type' content='text/html;charset=utf-8'>";
        body += "<head>";
        body += "<body>";
        body += "<h1>";
        body += std::to_string(rsp->_statu);
        body += " ";
        body += Util::StatusDesc(rsp->_statu);
        body += "</h1>";
        body += "</body>";
        body += "</html>";
        // 2.将页面数据当作响应正文，放入rsp中
        rsp->SetContent(body, "text/html");
    }
    // 将HttpResponse中的要素按照http协议格式进行组织，发送
    void WriteResponse(const Connection::PtrConnection &conn, const HttpRequest &req, HttpResponse &rsp)
    {
        // 1. 先完善头部字段
        // 连接字段
        if (req.Close() == true)
        {
            rsp.SetHeader("Connection", "close");
        }
        else
        {
            rsp.SetHeader("Connection", "keep-alive");
        }
        // 正文长度字段
        if (rsp._body.empty() == false && rsp.HasHeader("Content-Length") == false)
        {
            rsp.SetHeader("Content-Length", std::to_string(rsp._body.size()));
        }
        // 正文类型字段
        if (rsp._body.empty() == false && rsp.HasHeader("Content-Type") == false)
        {
            rsp.SetHeader("Content-Type", "application/octet-stream"); // 二进制流类型
        }
        // 重定向字段
        if (rsp._redirect_flag == true)
        {
            rsp.SetHeader("Location", rsp._redirect_url);
        }
        // 2.将rsp中的要素，按照http协议格式进行组织, 其实就是序列化
        std::stringstream rsp_str;
        rsp_str << req._version << " " << std::to_string(rsp._statu) << " " << Util::StatusDesc(rsp._statu) << "\r\n";
        for (auto &head : rsp._headers)
        {
            rsp_str << head.first << ": " << head.second << "\r\n";
        }
        rsp_str << "\r\n";
        rsp_str << rsp._body;
        // 3.发送数据
        conn->Send(rsp_str.str().c_str(), rsp_str.str().size());
    }
    // 判断是否是一个静态资源请求
    bool IsFileHandler(const HttpRequest &req)
    {
        // 1.必须设置了静态资源根目录
        if (_basedir.empty())
        {
            return false;
        }
        // 2.请求方法必须是GET或者HEAD请求方法
        if (req._method != "GET" && req._method != "HEAD")
        {
            return false;
        }
        // 3.请求的资源路径必须是合法的
        if (Util::ValidPath(req._path) == false)
        {
            return false;
        }
        // 4.请求的资源必须存在,且是一个普通文件
        // 有一种请求比较特殊---目录，请求的资源就是纯粹的相对根目录 /, 或者/image/ ，这种情况给后面默认追加一个index.html
        // 不要忘了前缀的相对根目录，要加上./wwwroot
        // 也就是将请求路径转化为实际存在的路径
        std::string req_path = _basedir + req._path;
        if (req_path.back() == '/')
        {
            req_path += "index.html";
        }
        if (Util::IsRegular(req_path) == false)
        {
            return false;
        }
        return true;
    }
    // 静态资源的请求处理 -- 将静态资源文件的数据读取出来，放到rsp的_body中，并设置content-type
    void FileHandler(const HttpRequest &req, HttpResponse *rsp)
    {
        std::string req_path = _basedir + req._path; // 将请求路径转化为实际存在的路径
        if (req_path.back() == '/')
        {
            req_path += "index.html";
        }
        bool ret = Util::ReadFile(req_path, &rsp->_body);
        if (ret == false)
        {
            return;
        }
        // 设置content-type
        std::string mime = Util::ExtMime(req_path);
        rsp->SetHeader("Content-Type", mime);
        return;
    }
    // 功能性请求的分类处理
    void Dispatcher(HttpRequest &req, HttpResponse *rsp, Handlers &handlers)
    {
        // 在对应的请求方法的路由表中，查找是否含有对应的资源请求的处理函数，有则调用，没有则返回404
        // 思想：路由表存储的键值对：--正则表达式 & 处理函数
        // 使用正则表达式，对请求的资源路径进行正则匹配，匹配成功，就使用对应函数进行处理
        for (auto &handler : handlers)
        {
            std::regex &re = handler.first;
            const Handler &functor = handler.second;
            bool ret = std::regex_match(req._path, req._matches, re);
            if (ret == false)
            {
                continue;
            }
            return functor(req, rsp); // 传入请求信息，和空的rsp，执行处理函数
        }
        rsp->_statu = 404; // 没有找到对应的处理函数，返回404
    }
    void Route(HttpRequest &req, HttpResponse *rsp)
    {
        // 1.对请求进行分辨，是一个静态请求还是一个功能性请求
        // 静态资源请求，则需要进行静态资源的处理
        // 功能性请求，则需要进行通过几个请求路由表来确定是否有处理函数
        // 既不是静态资源请求，也没有设置对应的功能性请求处理函数，就返回405
        // GET，HEAD，都默认先认为是静态资源请求
        if (IsFileHandler(req) == true)
        {
            // 是一个静态资源请求,进行静态资源的处理
            return FileHandler(req, rsp);
        }
        if (req._method == "GET" | req._method == "HEAD") // GET和HEAD的请求
            return Dispatcher(req, rsp, _get_route);
        else if (req._method == "POST")
            return Dispatcher(req, rsp, _post_route);
        else if (req._method == "PUT")
            return Dispatcher(req, rsp, _put_route);
        else if (req._method == "DELETE")
            return Dispatcher(req, rsp, _delete_route);

        rsp->_statu = 405; // 没有找到对应的处理函数，返回Method Not Allowed
        return;
    }
    // 连接建立的回调函数
    void OnConnected(const Connection::PtrConnection &conn) // 设置上下文
    {
        conn->SetContext(HttpContext());
        LOG(LogLevel::DEBUG) << "new connection : " << conn->Fd();
    }
    // 缓冲区数据解析 + 处理
    void OnMessage(const Connection::PtrConnection conn, Buffer *buf)
    {
        // 循环处理，因为说不定不止一个请求，可以连续处理多个请求
        while (buf->ReadableSize() > 0)
        {
            // 1.获取上下文
            HttpContext *context = conn->GetContext()->get<HttpContext>();
            // 2.通过上下文对缓冲区数据进行解析，得到HttpRequest对象
            // 如果缓冲区的数据解析出错，就直接恢复出错响应
            // 如果解析正常，且请求已经获取完毕，才开始去进行处理
            context->RecvHttpRequest(buf);
            HttpRequest &request = context->Request(); // 从上下文中提取出request
            HttpResponse rsp(context->ResponseStatu());
            if (context->ResponseStatu() >= 400) // 出错了
            {
                // 进行错误响应，关闭连接
                ErrorHandler(request, &rsp);       // 填充一个错误显示页面数据到rsp中
                WriteResponse(conn, request, rsp); // 组织响应，发送给客户端
                // 重置context的状态, 因为下面的ShutDown中会判断，如果接收缓冲区还有数据
                //还会来调用一遍这个函数，这个时候再去RecvHttpRequest处理的时候，就会因为context的状态不正确
                //一个处理函数都进不去，也就是缓冲区数据根本拿不出来，然后响应状态码也仍然不变
                // 那么就还会走到这里，然后ShutDown再判断缓冲区中有数据
                //这样就陷入了一个死循环，函数无限调用
                context->ReSet();
                buf->MoveReadOffset(buf->ReadableSize()); // 出错了就把缓冲区数据清空，保险起见
                conn->ShutDown();                  // 关闭连接
                return;
            }
            if (context->RecvStatu() != RECV_HTTP_OVER)
            {
                // 当前请求还没有接收完整
                return; // 则退出，等新数据到来再重新继续处理
            }
            // 3.请求路由 + 业务处理
            Route(request, &rsp);

            // 4.对HttpResponse进行组织发送
            WriteResponse(conn, request, rsp); // 组织响应，发送给客户端

            // 5.重置上下文
            context->ReSet(); // 避免影响下一条请求的处理

            // 6.根据长短连接判断是否关闭连接或者继续处理
            if (rsp.Close() == true)
                conn->ShutDown(); // 短连接则直接关闭
        }
    }

private:
    Handlers _get_route;
    Handlers _post_route;
    Handlers _put_route;
    Handlers _delete_route;
    std::string _basedir; // 静态资源根目录
    TcpServer _server;
};